package com.example.demo.config.ftpclient;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
public class FTPUtil {
    /** 本地字符编码 */
    private static String LOCAL_CHARSET = "GBK";
    // FTP协议里面，规定文件名编码为iso-8859-1
    private static String SERVER_CHARSET = "ISO-8859-1";

    /**
     * Description: 向FTP服务器上传文件
     * @param url FTP服务器hostname
     * @param port FTP服务器端口
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param path FTP服务器保存目录
     * @param filename 上传到FTP服务器上的文件名
     * @param input 输入流
     * @return 成功返回true，否则返回false
     */
    public static boolean uploadFile(String url, int port,
                                     String username, String password,
                                     String path, String filename,
                                     InputStream input) {
        boolean success = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(url, port);//连接FTP服务器
            //如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
            ftp.login(username, password);//登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return success;
            }
            ftp.changeWorkingDirectory(path);
            //上传文件时，文件名称需要做编码转换
            filename = new String(filename.getBytes(LOCAL_CHARSET),
                    SERVER_CHARSET);

            ftp.storeFile(filename, input);

            input.close();
            ftp.logout();
            success = true;
        } catch (IOException e) {
            log.error(e.getMessage());
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                    log.error(ioe.getMessage());
                }
            }
        }
        return success;
    }

    /**
     * Description: 从FTP服务器下载文件
     * @param url FTP服务器hostname
     * @param port FTP服务器端口
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param remotePath FTP服务器上的相对路径
     * @param fileName 要下载的文件名
     * @param localPath 下载后保存到本地的路径
     * @return
     */
    public static boolean downFile(String url, int port,
                                   String username, String password,
                                   String remotePath, String fileName,
                                   String localPath) {
        boolean success = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(url, port);
            //如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
            ftp.login(username, password);//登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return success;
            }
            ftp.changeWorkingDirectory(remotePath);//转移到FTP服务器目录
            FTPFile[] fs = ftp.listFiles();
            for(FTPFile ff:fs){
                if(ff.getName().equals(fileName)){
                    File localFile = new File(localPath+"/"+ff.getName());

                    OutputStream is = new FileOutputStream(localFile);
                    ftp.retrieveFile(ff.getName(), is);
                    is.close();
                }
            }

            ftp.logout();
            success = true;
        } catch (IOException e) {
            log.error(e.getMessage());
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                    log.error(ioe.getMessage());
                }
            }
        }
        return success;
    }


    /**
     * Description: 从FTP服务器下载文件
     * @param url FTP服务器hostname
     * @param port FTP服务器端口
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param filePaths FTP服务器上的文件相对路径(如：BPMN/purchasingflow.bpmn20.xml)的集合
     * @return
     */
    public static List<InputStream> fileInputStream(String url, int port,
                                   String username, String password,
                                   List<String> filePaths) {
        if (filePaths == null || filePaths.size() < 1){
            return null;
        }
        List<InputStream> inputStreams = null;
        FTPClient ftpClient = new FTPClient();
        try {
            inputStreams = new ArrayList<>();
            int reply;
            ftpClient.connect(url, port);
            //如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
            ftpClient.login(username, password);//登录
            reply = ftpClient.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftpClient.disconnect();
                return null;
            }
            if (FTPReply.isPositiveCompletion(ftpClient.sendCommand(
                    "OPTS UTF8", "ON"))) {// 开启服务器对UTF-8的支持，如果服务器支持就用UTF-8编码，否则就使用本地编码（GBK）.
                LOCAL_CHARSET = "UTF-8";
            }
            ftpClient.setControlEncoding(LOCAL_CHARSET);

            Map<String, List<String>> fileMap = new HashMap<String, List<String>>();

            //相同路径的文件堆放在一起
            for (String filePath : filePaths) {
                String[] strings = filePath.split("/");
                //获取文件名
                String fileName = strings[strings.length - 1];
                //获取文件路径
                String parentPath = strings.length > 1 ?
                        filePath.replace(fileName,"").trim() : fileName;

                if (!fileMap.containsKey(parentPath))
                    fileMap.put(parentPath, new ArrayList<String>());
                fileMap.get(parentPath).add(fileName);
            }

            for (String parentPath : fileMap.keySet()) {
                //转移到FTP服务器目录
                ftpClient.changeWorkingDirectory(parentPath);
                for (String fileName : fileMap.get(parentPath)) {
                    inputStreams.add(ftpClient.retrieveFileStream(fileName));
//                    inputStreamMap.put(parentPath+fileName, ftpClient.retrieveFileStream(fileName));
                }
            }

            ftpClient.logout();
        } catch (IOException e) {
            log.error(e.getMessage());
        } finally {
            if (ftpClient.isConnected()) {
                try {
                    ftpClient.disconnect();
                } catch (IOException ioe) {
                    log.error(ioe.getMessage());
                }
            }
        }
        return inputStreams;
    }

}
